#include "../logs/lcglog.h"
#include <unistd.h>
/* 拓展一个以时间作为成为文件滚动切换类型的日志落地模块
    1.以时间进行文件滚动，实际上是以时间段进行文件滚动
        实现思想：以当前系统时间，取模时间段大小，可以得到当前时间段是第几个时间段，
        每次与当前系统时间取模，判断与当前文件的时间段是否一致，不一致就代表不是一个时间段，就切换文件
*/
/*落地拓展：以时间作为滚动的依据的滚动文件的落地方式*/
enum class TimeGap{
    GAP_SECOND,
    GAP_MINUTE,
    GAP_HOUR,
    GAP_DAY
};

  //落地方向：滚动文件(以时间进行滚动)
    class RollByTimeSink : public lcglog::LogSink{
        public:
            //构造时将文件名传入，并且打开文件，将操作句柄管理起来，通过文件句柄管理文件
            RollByTimeSink(const std::string& pathname, TimeGap gap_type):_pathname(pathname),
            _cur_gap(0){
                switch(gap_type){
                    case TimeGap::GAP_SECOND: _gap_size = 1;    break;
                    case TimeGap::GAP_MINUTE: _gap_size = 60;   break;
                    case TimeGap::GAP_HOUR: _gap_size = 3600;   break;
                    case TimeGap::GAP_DAY: _gap_size = 3600*24; break;
                }
                _cur_gap = (_gap_size == 1) ? lcglog::util::Date::now() : lcglog::util::Date::now() % _gap_size;//获取当前是第几个时间段
                std::string filename = createNewFile();
                lcglog::util::File::createDirectory(lcglog::util::File::path(filename));
                _ofs.open(filename, std::ios::binary | std::ios::app);
                assert(_ofs.is_open());
            }
            
            //将日志消息写入到指定文件--判断当前时间是否是当前文件的时间段，不是就切换文件
            void log(const char* data, size_t len){
                time_t cur = lcglog::util::Date::now();//获取系统当前时间
                if((cur % _gap_size) != _cur_gap){
                    _ofs.close();//关闭原来已经打开的文件
                
                    std::string filename = createNewFile();
                    _ofs.open(filename, std::ios::binary | std::ios::app);
                    assert(_ofs.is_open());
                }

                _ofs.write(data, len);
                assert(_ofs.good());
            }
        private:
            //进行大小判断，超过指定大小则创建新文件
            std::string createNewFile(){
                time_t t = lcglog::util::Date::now();
                struct tm lt;
                localtime_r(&t, &lt);
                std::stringstream filename;
                filename << _pathname;
                filename << lt.tm_year + 1900;
                filename << lt.tm_mon + 1;
                filename << lt.tm_mday;
                filename << lt.tm_hour;
                filename << lt.tm_min;
                filename << lt.tm_sec;
                filename << ".log";

                return filename.str();
            }
        private:
            //通过基础文件名+拓展文件名（以时间生成）,组成一个实际的当前输出文件名
            std::string _pathname;
            std::ofstream _ofs; //操作句柄
            size_t _cur_gap;//当前是第几个时间段
            size_t _gap_size;//时间段的大小
    };
int main()
{
    /*正常现象：我要创建5个文件，但实际创建了6个文件，多了一个空文件，并且写入文件的数据数量不足1000条*/
    /*最后一个文件是没有输出的，跳出了，但文件是已经创建了*/
    /*并且这时按照时间进行滚动来创建文件的，在规定的时间内写文件的数据不一定达到了指定的数量*/
    std::unique_ptr<lcglog::LoggerBuilder> builder(new lcglog::GlobalLoggerBuilder());
    builder->buildLoggerName("async_logger");
    builder->buildLoggerLevel(lcglog::LogLevel::value::WARN);
    builder->buildLoggerFormatter("[%c][%f:%l]%m%n");
    builder->buildLoggerType(lcglog::LoggerType::LOGGER_ASYNC);
    builder->buildSink<RollByTimeSink>("./logfile/async-roll-by-time", TimeGap::GAP_SECOND);
    lcglog::Logger::ptr logger = builder->build();

    size_t old = lcglog::util::Date::now();
    while(lcglog::util::Date::now() < old + 5){
        logger->fatal("这是一条测试日志！");
        usleep(1000);
    }

                        
    return 0;
}